34f6b0ce75886c2ff5f759d916e82c5beb6fb982,nd4j-backends/nd4j-backend-impls/nd4j-cuda/src/main/java/org/nd4j/linalg/jcublas/ops/executioner/CudaExecutioner.java,CudaExecutioner,exec,#IndexAccumulation#number#,588
Before Change
}
INDArray ret = null;
if (op.zeroDouble() > -0.01f && op.zeroDouble() < 0.01f) {
ret = Nd4j.zeros(retShape);
} else {
ret = Nd4j.valueArrayOf(retShape, op.zeroDouble());
After Change
}
INDArray ret = null;
if (0.0 + Math.abs(op.zeroDouble()) <= Nd4j.EPS_THRESHOLD) {
ret = Nd4j.zeros(retShape);
} else {
if (op.x().data().dataType() == DataBuffer.Type.DOUBLE)
ret = Nd4j.valueArrayOf(retShape,op.zeroDouble());
else if (op.x().data().dataType() == DataBuffer.Type.FLOAT)
ret = Nd4j.valueArrayOf(retShape,op.zeroFloat());
else if (op.x().data().dataType() == DataBuffer.Type.HALF)
ret = Nd4j.valueArrayOf(retShape,op.zeroHalf());
}
op.setZ(ret);